from .LeNet import LeNet5
from .AlexNet import AlexNet
from .vgg16 import vgg16
from .squeezenet import SqueezeNet
from .mobilenetv2 import mobilenet_v2
from .mobilenetv3 import mobilenetv3
from .shufflenet_v2 import shufflenetv2
from .ghostnet import ghostnet
from .densenet import DenseNet
from .resnet50 import resnet50
from .resnext import resnext50





get_model_from_name = {
    "LeNet"         : LeNet5,
    "AlexNet"       : AlexNet,

    "vgg16"         : vgg16,

    "squeezenet"    : SqueezeNet,
    "mobilenetv2"   : mobilenet_v2,
    "mobilenetv3"   : mobilenetv3,
    "shufflenetv2"  : shufflenetv2,
    "ghostnet"      : ghostnet,

    "DenseNet"      : DenseNet,
    "resnet50"      : resnet50,
    "resnext50"     : resnext50,
}
